package com.github.yangyishe.p300;

import com.github.yangyishe.TreeNode;

/**
 * 222. 完全二叉树的节点个数
 * https://leetcode.cn/problems/count-complete-tree-nodes/?envType=study-plan-v2&envId=top-interview-150
 *
 * 给你一棵 完全二叉树 的根节点 root ，求出该树的节点个数。
 *
 * 完全二叉树 的定义如下：在完全二叉树中，除了最底层节点可能没填满外，其余每层节点数都达到最大值，并且最下面一层的节点都集中在该层最左边的若干位置。若最底层为第 h 层，则该层包含 1~ 2h 个节点。
 *
 *
 *
 * 示例 1：
 *
 *
 * 输入：root = [1,2,3,4,5,6]
 * 输出：6
 * 示例 2：
 *
 * 输入：root = []
 * 输出：0
 * 示例 3：
 *
 * 输入：root = [1]
 * 输出：1
 *
 *
 * 提示：
 *
 * 树中节点的数目范围是[0, 5 * 104]
 * 0 <= Node.val <= 5 * 104
 * 题目数据保证输入的树是 完全二叉树
 *
 *
 * 进阶：遍历树来统计节点是一种时间复杂度为 O(n) 的简单解决方案。你可以设计一个更快的算法吗？
 */
public class Problem222 {
    public static void main(String[] args) {
        Integer[] root=new Integer[]{1,2,3,4,5,6};
        TreeNode instanceNew = TreeNode.getInstanceNew(root);

        Problem222 problem222 = new Problem222();
        int count = problem222.countNodes(instanceNew);

        System.out.println(count);
    }

    /**
     * 思路:
     * 考虑使用类似二分查找的方法. 确定层级的数量, 及最后一个层级的节点数目.
     * 如果确定最大层级为n, 最后一个层级的节点数目为m
     * 则总节点数量为2^(n-1)-1+m
     *
     * 先遍历最左节点深度, 以确定最大层级数
     * 在遍历右子树的最左节点深度, 是否等于最大层级, 如果等于, 则继续在右子树的右子树中找最左节点深度, 否则, 找左子树的右子树的最左节点深度
     *
     * @param root
     * @return
     */
    public int countNodes(TreeNode root) {
        if(root==null){
            return 0;
        }

        return countNodes(root.left)+countNodes(root.right)+1;
    }


//    public int maxLevel(TreeNode node){
//        if(node.left==null){
//            return 1;
//        }else{
//            return maxLevel(node.left)+1;
//        }
//    }
//
//    public int LastLevelNodeCount(TreeNode node,int maxLevel){
//        if(node==null){
//            return 1;
//        }
//        int rightMaxLevel = maxLevel(node.right);
//        if(rightMaxLevel!=maxLevel){
//            return LastLevelNodeCount(node.left,maxLevel-1);
//        }else{
//            return LastLevelNodeCount(node.right,maxLevel-1);
//        }
//    }
}
